//给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。 
//
// 
//
// 示例 1： 
//
// 
//
// 
//输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
//输出：6
//解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。 
// 
//
// 示例 2： 
//
// 
//输入：height = [4,2,0,3,2,5]
//输出：9
// 
//
// 
//
// 提示： 
//
// 
// n == height.length 
// 0 <= n <= 3 * 104 
// 0 <= height[i] <= 105 
// 
// Related Topics 栈 数组 双指针 动态规划 
// 👍 2433 👎 0


package service.week01.leetcode.editor.cn;

import java.util.Stack;

//Java：接雨水
public class P42TrappingRainWater {
    public static void main(String[] args) {
        Solution solution = new P42TrappingRainWater().new Solution();
        // TO TEST
        System.out.println(solution.trap(new int[]{0,1,0,2,1,0,1,3,2,1,2,1}));
    }

    //leetcode submit region begin(Prohibit modification and deletion)
    class Solution {
        /**
         * 最近相关性
         *
         * @param height
         * @return
         */
        public int trap(int[] height) {
            //定义结果
            int res = 0;
            Stack<Rect> stack = new Stack<>();
            Rect rect = new Rect();
            rect.height = 0;
            rect.width = 0;
            stack.push(rect);
            for (int h : height) {
                int width = 0;
                //单调栈
                while (stack.size() > 1 && stack.peek().height <= h) {
                    width += stack.peek().width;
                    int bottom = stack.peek().height;
                    stack.pop();
                    //计算
                    res += Math.max(0, Math.min(stack.peek().height, h) - bottom) * width;
                }
                Rect r = new Rect();
                r.height = h;
                r.width = width + 1;
                stack.push(r);
            }
            return res;
        }

        private class Rect {
            int height;
            int width;
        }
    }
//leetcode submit region end(Prohibit modification and deletion)

}